D:\a\tools.proto\tools.proto\cli\src\builder.rs
Line | Count | Source |
1 | | // Copyright (c) 2024, BlockProject 3D |
2 | | // |
3 | | // All rights reserved. |
4 | | // |
5 | | // Redistribution and use in source and binary forms, with or without modification, |
6 | | // are permitted provided that the following conditions are met: |
7 | | // |
8 | | // * Redistributions of source code must retain the above copyright notice, |
9 | | // this list of conditions and the following disclaimer. |
10 | | // * Redistributions in binary form must reproduce the above copyright notice, |
11 | | // this list of conditions and the following disclaimer in the documentation |
12 | | // and/or other materials provided with the distribution. |
13 | | // * Neither the name of BlockProject 3D nor the names of its contributors |
14 | | // may be used to endorse or promote products derived from this software |
15 | | // without specific prior written permission. |
16 | | // |
17 | | // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
18 | | // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
19 | | // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
20 | | // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR |
21 | | // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
22 | | // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
23 | | // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
24 | | // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
25 | | // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING |
26 | | // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
27 | | // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
28 | | |
29 | | use crate::Args; |
30 | | use bp3d_protoc::api::core::generator::{Generator, Params}; |
31 | | use bp3d_protoc::api::core::loader::Loader; |
32 | | use bp3d_protoc::compiler::util::imports::{ImportSolver, ProtocolStore}; |
33 | | use bp3d_util::result::ResultExt; |
34 | | use std::path::Path; |
35 | | |
36 | | pub struct Builder<'a, I, G> { |
37 | | pub protocols: ProtocolStore<'a, I>, |
38 | | pub generator: Generator<'a, G>, |
39 | | pub params: Params, |
40 | | } |
41 | | |
42 | | impl<'a, I: ImportSolver, G: bp3d_protoc::gen::Generator> Builder<'a, I, G> { |
43 | 0 | pub fn new(loader: Loader, args: &'a Args, solver: &'a I, generator: G) -> Self { |
44 | 0 | let protocols = loader.compile(solver).expect_exit("failed to compile protocols", 1); |
45 | 0 | let params = if let Some(features) = &args.features { Branch (45:29): [Folded - Ignored]
|
46 | 0 | let mut params = Params::new(); |
47 | 0 | for f in features { |
48 | 0 | f.apply(&mut params); |
49 | 0 | } |
50 | 0 | params |
51 | | } else { |
52 | 0 | Params::default() |
53 | | }; |
54 | 0 | let output = args.output.as_deref().unwrap_or(Path::new("./")); |
55 | 0 | let mut generator = Generator::with_default_codecs(output, generator); |
56 | 0 | if let Some(file_header) = &args.file_header { Branch (56:16): [Folded - Ignored]
|
57 | 0 | generator.set_file_header(file_header); |
58 | 0 | } |
59 | 0 | Self { |
60 | 0 | protocols, |
61 | 0 | generator, |
62 | 0 | params, |
63 | 0 | } |
64 | 0 | } |
65 | | } |